import os

def save_filenames_to_txt(folder_path, txt_file_path):
    """
    将指定文件夹中的所有文件的文件名前缀保存到txt文件中。

    :param folder_path: 文件夹的路径。
    :param txt_file_path: 要保存的txt文件的路径。
    """
    # 确保文件夹路径有效
    if not os.path.isdir(folder_path):
        print("指定的文件夹不存在")
        return

    # 获取文件夹中的所有文件名
    filenames = os.listdir(folder_path)

    # 打开txt文件准备写入
    with open(txt_file_path, 'w', encoding='utf-8') as file:
        for filename in filenames:
            # 分离文件名和扩展名，只获取文件名前缀
            prefix = os.path.splitext(filename)[0]
            file.write(prefix + '\n')

    print(f"文件名前缀已保存到 {txt_file_path}")

# 使用示例
folder_path = r'F:\Dataset\sumdata(yes)\SumData\images\test'  # 替换为你的文件夹路径
txt_file_path = 'test.txt'  # 你想保存的txt文件名
save_filenames_to_txt(folder_path, txt_file_path)
